CoWindow Attention: Full Causal Coverage Is a Collective Property
CoWindow 注意力:全因果覆盖是头集合的集体属性
🧭 LLM 稀疏注意力 · 集体因果覆盖 ★ 深度解读
The Hong Kong University of Science and Technology (Guangzhou) · Beijing Academy of Artificial Intelligence · Université Paris Cité
用 KV 头互补长程窗口实现全因果覆盖,训练/解码延迟降 7.4x/3x 且质量接近 FullAttn
📌 工作参考价值(4 点)
- 长上下文 LLM 用于用户行为轨迹、会话日志或 Agent 工具调用历史时,可考虑把 FullAttn 换成 CoWA 式结构:近邻窗口保留最近行为/当前会话,prefix-sink 固定放 user profile 或 system prompt,远处历史按 KV head 切分互补窗口,训练和推理都能省显存省 FLOPs,同时不牺牲关键关联召回。
- 静态位置窗口比动态 router/indexer 更适合线上:不需要额外检索选择,避免了线上 TopK 漂移和 router 延迟;如果业务里做长序列检索式注意力,可以借鉴 position-defined sparsity + collective coverage,把长程访问做成确定性分片,工程实现更稳。
- 多 GPU 部署时按全局 KV head 索引分配窗口,与 tensor parallelism 对齐,每个 rank 只持有互补窗口的 KV cache,可降低长上下文推理显存和 KV cache 访问;适合商品描述生成、Agent 多步轨迹分析等长上下文服务降本。
- 用 synthetic associative recall 快速评估长序列建模是否保留 item 关联,比直接跑长上下文 benchmark 更便宜,适合在推荐系统用户行为序列建模上线前做快速 ablation。